import os

import cherrypy
from cherrypy.test import helper

curdir = os.path.join(os.getcwd(), os.path.dirname(__file__))


class VirtualHostTest(helper.CPWebCase):
    @staticmethod
    def setup_server():
        class Root:
            @cherrypy.expose
            def index(self):
                return 'Hello, world'

            @cherrypy.expose
            def dom4(self):
                return 'Under construction'

            @cherrypy.expose
            def method(self, value):
                return 'You sent %s' % value

        class VHost:
            def __init__(self, sitename):
                self.sitename = sitename

            @cherrypy.expose
            def index(self):
                return 'Welcome to %s' % self.sitename

            @cherrypy.expose
            def vmethod(self, value):
                return 'You sent %s' % value

            @cherrypy.expose
            def url(self):
                return cherrypy.url('nextpage')

            # Test static as a handler (section must NOT include vhost prefix)
            static = cherrypy.tools.staticdir.handler(
                section='/static',
                dir=curdir,
            )

        root = Root()
        root.mydom2 = VHost('Domain 2')
        root.mydom3 = VHost('Domain 3')
        hostmap = {
            'www.mydom2.com': '/mydom2',
            'www.mydom3.com': '/mydom3',
            'www.mydom4.com': '/dom4',
        }
        cherrypy.tree.mount(
            root,
            config={
                '/': {
                    'request.dispatch': cherrypy.dispatch.VirtualHost(
                        **hostmap,
                    ),
                },
                # Test static in config (section must include vhost prefix)
                '/mydom2/static2': {
                    'tools.staticdir.on': True,
                    'tools.staticdir.root': curdir,
                    'tools.staticdir.dir': 'static',
                    'tools.staticdir.index': 'index.html',
                },
            },
        )

    def testVirtualHost(self):
        self.getPage('/', [('Host', 'www.mydom1.com')])
        self.assertBody('Hello, world')
        self.getPage('/mydom2/', [('Host', 'www.mydom1.com')])
        self.assertBody('Welcome to Domain 2')

        self.getPage('/', [('Host', 'www.mydom2.com')])
        self.assertBody('Welcome to Domain 2')
        self.getPage('/', [('Host', 'www.mydom3.com')])
        self.assertBody('Welcome to Domain 3')
        self.getPage('/', [('Host', 'www.mydom4.com')])
        self.assertBody('Under construction')

        # Test GET, POST, and positional params
        self.getPage('/method?value=root')
        self.assertBody('You sent root')
        self.getPage('/vmethod?value=dom2+GET', [('Host', 'www.mydom2.com')])
        self.assertBody('You sent dom2 GET')
        self.getPage(
            '/vmethod',
            [('Host', 'www.mydom3.com')],
            method='POST',
            body='value=dom3+POST',
        )
        self.assertBody('You sent dom3 POST')
        self.getPage('/vmethod/pos', [('Host', 'www.mydom3.com')])
        self.assertBody('You sent pos')

        # Test that cherrypy.url uses the browser url, not the virtual url
        self.getPage('/url', [('Host', 'www.mydom2.com')])
        self.assertBody('%s://www.mydom2.com/nextpage' % self.scheme)

    def test_VHost_plus_Static(self):
        # Test static as a handler
        self.getPage('/static/style.css', [('Host', 'www.mydom2.com')])
        self.assertStatus('200 OK')
        self.assertHeader('Content-Type', 'text/css;charset=utf-8')

        # Test static in config
        self.getPage('/static2/dirback.jpg', [('Host', 'www.mydom2.com')])
        self.assertStatus('200 OK')
        self.assertHeaderIn('Content-Type', ['image/jpeg', 'image/pjpeg'])

        # Test static config with "index" arg
        self.getPage('/static2/', [('Host', 'www.mydom2.com')])
        self.assertStatus('200 OK')
        self.assertBody('Hello, world\r\n')
        # Since tools.trailing_slash is on by default, this should redirect
        self.getPage('/static2', [('Host', 'www.mydom2.com')])
        self.assertStatus(301)
